package com.cold.oss.service.impl;

import com.aliyun.oss.OSSClient;
import com.aliyun.oss.model.CannedAccessControlList;
import com.cold.oss.service.OssService;
import com.cold.oss.utils.ConstantPropertiesUtils;
import com.cold.servicebase.exceptionhandler.GuliException;
import org.joda.time.DateTime;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.util.UUID;

/**
 * @author cold_and_cold
 * @create 2021-01-27 16:37
 */
@Service
public class OssServiceImpl implements OssService {
    @Override
    public String uploadFileAvatar(MultipartFile file) {
        //获取阿里云存储相关常量
        String endPoint = ConstantPropertiesUtils.END_POINT;
        String accessKeyId = ConstantPropertiesUtils.ACCESS_KEY_ID;
        String accessKeySecret = ConstantPropertiesUtils.ACCESS_KEY_SECRET;
        String bucketName = ConstantPropertiesUtils.BUCKET_NAME;

        String uploadUrl = null;

        try {
                //判断oss实例是否存在：如果不存在则创建，如果存在则获取
                OSSClient ossClient = new OSSClient(endPoint, accessKeyId, accessKeySecret);
                if (!ossClient.doesBucketExist(bucketName)) {
                    //创建bucket
                    ossClient.createBucket(bucketName);
                    //设置oss实例的访问权限：公共读
                    ossClient.setBucketAcl(bucketName, CannedAccessControlList.PublicRead);
                }
                //获取上传文件流
                InputStream inputStream = file.getInputStream();

            //文件名：uuid.扩展名
                String fileName = file.getOriginalFilename();
                String uuid = UUID.randomUUID().toString().replaceAll("-", "");
                fileName = uuid+fileName;
                String dataPath = new DateTime().toString("yyyy/MM/dd");
                fileName = dataPath + fileName;
            //文件上传至阿里云
                ossClient.putObject(bucketName, fileName, inputStream);

            // 关闭OSSClient。
                ossClient.shutdown();

            //获取url地址
                uploadUrl = "https://" + bucketName + "." + endPoint + "/" + fileName;
                return  uploadUrl;

        } catch (IOException e) {
            e.printStackTrace();
        }

        return null;
    }
}
